HLO Feature Dataset for AI Resource Estimation

A dataset designed to support AI-driven resource estimation like runtime prediction to support HPC scheduling optimization by leveraging compiler-level High-Level Optimizer (HLO) graph features and deep learning workload metadata.

References

• 📂 Hugging Face Dataset: ICICLE-AI/ResourceEstimation_HLOGenCNN
• 📓 Baseline Notebook: XGBoost Resource Estimation
• 🛠️ TensorFlow XLA Compiler: XLA Docs
• 📚 XGBoost Library: XGBoost Documentation

---

Acknowledgements

National Science Foundation (NSF) funded AI institute for Intelligent Cyberinfrastructure with Computational Learning in the Environment (ICICLE) (OAC 2112606)